How to implement market basket analysis in Python